package leetcode;

/**
 * @author caifangyi
 * @date 2022/8/1
 */

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Deque;
import java.util.List;

/**
 * 1161. 最大层内元素和
 * 
 * 给你一个二叉树的根节点root。设根节点位于二叉树的第 1 层，而根节点的子节点位于第 2 层，依此类推。
 *
 * 请返回层内元素之和 最大 的那几层（可能只有一层）的层号，并返回其中最小 的那个。
 *
 * 
 *
 * 示例 1：
 *
 *
 *
 * 输入：root = [1,7,0,7,-8,null,null]
 * 输出：2
 * 解释：
 * 第 1 层各元素之和为 1，
 * 第 2 层各元素之和为 7 + 0 = 7，
 * 第 3 层各元素之和为 7 + -8 = -1，
 * 所以我们返回第 2 层的层号，它的层内元素之和最大。
 * 示例 2：
 *
 * 输入：root = [989,null,10250,98693,-89388,null,null,null,-32127]
 * 输出：2
 * 
 *
 * 提示：
 *
 * 树中的节点数在[1, 104]范围内
 * -105<= Node.val <= 105
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/maximum-level-sum-of-a-binary-tree
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Day1161 {
    public static void main(String[] args) {

    }

    /**
     * Definition for a binary tree node.
     *  */
     public static class TreeNode {
         int val;
         TreeNode left;
         TreeNode right;
         TreeNode() {}
         TreeNode(int val) { this.val = val; }
         TreeNode(int val, TreeNode left, TreeNode right) {
             this.val = val;
             this.left = left;
             this.right = right;
         }
    }

    class Solution {
        public int maxLevelSum(TreeNode root) {
            //树转队列
            Deque<TreeNode> deque = new ArrayDeque<>();
            deque.add(root);
            int depth = 1;
            int maxSum = -1000000;
            int maxDepth = 0;
            while(!deque.isEmpty()){
                int end = deque.size();
                int sum = 0;
                int index = 0;
                while(index < end) {
                    root = deque.pop();
                    index++;
                    sum += root.val;
                    if(root.left != null){
                        deque.add(root.left);
                    }
                    if(root.right!=null){
                        deque.add(root.right);
                    }
                }
                if(maxSum < sum){
                    maxSum = sum;
                    maxDepth = depth;
                }
                depth ++;

            }

            return maxDepth;
        }

    }
}
